Added controller with save functionality