Improved the output of the program

Also made the arguments behave better