toy experiment update
minor changes, tracking while training