Fixed a couple polish things