Minor changes made during class