Tidy code, add nav bar