Removed hard-coded argument and added new argparse argument to ensure that ylimit can be easily specified to the user