Added part b of problem
finalized for lab pc testing