Small script restyling