refactor: reorganize code around to shoosh compiler