Cleaned up JS and animations