just cleaning up the code