Solution assignment 2