HW11 AI commit: 
Fix the auto scale functionality.