neaten the code
use graph instead of ufWorkGraph