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