part one done
uncommented last part and removed print statements