python draft for cs5100 assign3