Added basic save functionality