added predict class