Restructured
Clean up directory
Cleaned up feature_selection
Adding feature indices to data.py
Mask numpy array
added the prediction model