Updated code to compile with ProB 2.0