Fix formating of code