code climate suggestions corrected