Complete assignment 1