done with problem two and unit tested