improving code styling

There are some brackets missing