use argparse for input
Merge pull request #1 from MaZderMind/argparse

use argparse for input