sorted out most bugs, model compiles, working on training loop