Remove unncessary part in models. Load models automatically