python assignment 2