added nonlinearities to reconstructive model, evaluating and plotting losses improved