Cleaning up JS exercise