Python lab using classes to represent objects