Remove unncessary part in models