[IMPROVEMENT] small update on style