Complete assignment 2