prettying the code a little :)