Made model after important features only