Assinging a Sum in Python