Adding first part of homework 1