add python assignment 3