Args cleanup

I made some cleanup on the args parsing. Now It's much more pythonic.