First draft of assignment 3