many fixes and added calculation of feature importances at classify stage. also moved set up of masks into __init__