added the framework for the second part of the homework to test in