Added display classes and fixed the jQuery menu