Made change when stephen pointed out that what I had shouldn't work, this way is more *correct* WKB