plotting sqrt(Bu) instead of Bu and some bug fixes