Heiken Ashi Colored Moving Average

This indicator is meant to plot a moving average but the color of the moving average will change based on Heikin Ashi. Its seems to be slightly off, I would love any suggestions on improving this indicator.

Thanks