Clean unnecessary part/ initial predict func