REFACTORED: 1.button not working, 2.need to write to file or store to db