Fixd up the corner detection a bit?