Re-added ub_test: Code not working becuase of changes to task struct - need to fix