cleaned up the code
cleaned up code