fix code layout