Fixed a few small things with arguments