updated plotting in all_functions