Did I branch correctly?
Fixed the overflow error. Remeber to convert image to float