Got the diagonal working