refined nodes, added color and correct connections