Assignment 2: done