Added homework 2 and tests