Added proper argument parsing for the python script