Better handling for run_vasp.py