Fix the sudden colour change

Fix the sudden colour change in the generate image of the simulator. This can be done by adding colour bar limit.