Start working on diagram mode
Added diagram mode