added homework 2.1 solution