Homework 6 Update

Put scaling back in