Added diagonal to fix dimension problem