Update to the codes

So I handled at the commands in a different way, that allows me to work
easier on the program's code and how the arguments are handled.
minor correction