Done with UI logic, need file persistence..