main: Add proper argument parsing