Beginning of homework 2