fix code prettyness