Cleaned up styling and confusing parts