Added controller logic, doesnt save