Solution to Assignment 2 +more comments