More edge detection

TODO: Fix range in loop