Update

Cleaned up some code