Add solutions for Assignment 4