Complete assignment 4