Add input parsing and data storage in main method