bonus exercise, want to clean up the if logic, get to it later