Addiing solution to 2.1