timeevolution works for square now