cleaned a code a little bit