Updated my rectangle code