Fixing my code before anyone looks