python assignment 1