Cleaned some code

Taking a small break