add color class and fix geometry rendering