clean the code a little