cleaned up the code