Clean up JavaScript and styling