Update Assignmant2.py

i finished Ground Floor part