Allow blocks that are not squares to be straighten as well