Solved small issue with colors, created a mini script for data generation