Working on adding predicates into stubbing