Adds screen argument to rw_randomize.py