Added curly brackets that caused error

Curly brackets addition