Model improvements

Introduced feature extraction, remembering best model and stopping early in case of no significant improvements for 3 epochs. Removed usage of summary_time and updated code base accordingly.