fixing more code climate