Working on RGB dataset