Updated with colors

Not very pretty but hope it works