Improved responsive design and cleaned up js