Update

Cleaned up code and cleaner styling.