further work on assignment 1