Finish Assignment 4

Only finish the required task. Extra credit is for further development.