Added utility functions in `util`. And an example training file with model example in `train.py`