finished the first part of the assignment!!