Clean up code
Use NSUserDefaults for storing data