started on homework 1, adding all the functions, basic tests