Python lab 3 undone