edited my changes to make this assignment correct