Added ugly color plot implementation
Fixed color plots