Code working and brief test written