work on diagram