draft of month 2 assignment