Update lab0.py

part one done