Complete assignment 5