Cleaned up UI, including adding a back button on the note view