Swapped out function assertions in favour of ArgumentParser