still buggy branch lab 4
Completed Lab 4