cleaned up the code a bit

just cleaned up the code