Update

cleaned up code