Wrote first code for Ass3.Prob3