Code cleaned, added function for readability.

Important methods: regression & getFeatures