New features

Tidied code - added back button