Add draft of assignment 5