Problem 3 done :D