edited code styling