Your modifications sound good. Please visit us at our office hours on Mondays from 7-9pm if you would like help coding!