add logic stages graph