- Added main function with mode selection;
- Added code to training and testing modes;
TODO:
- Make the training mode model persistent (sqlite)