fix bug, and adjust plot colours