Added first round solution to Problem 4