Change references. Added agent visualization. Started train implementation