updated code styling