Cleaned up the code

it still works