improved BAIT plotting function