homework 5 partial working

Missing unit test and graph