edited code style