first working version with colors