Update

Cleaned up code.