Added UI and basic save mechanism