Edge add working now
Improved node shapes