Add basic mobile nav
Improve mobile nav JS