Cleaned up some code
...
...
...