Feature/coloring

Should work once the neighbors are being set