Added Command line arguments and cleaned code a bit