Fix back button, further styling