Add argument parsing for level selection