Refactoring compute output function
Bug fixing: pep8 and pylint corrections
Refactoring to decrease the argument number in function
Refactoring prediction methods
Bug fixing: models in batch_predict, test_set_headers, and file close
Pep8 and pylint cleaning
Updating version to 0.2.1