further work on assignment 2