1. Refactored the code.
2. Exposed Prediction and Build Model as Service.