removed deprecated heatmap function & added loop_direction option to argparse