still working on assignment 4