Updated with my code

first draft of the assignment