Added first iteration of lab4.py'