Small cleanup, fixes some very minor mistakes