Cleaned code

:P