Small style cleanup