Messed around with the code to eliminate bugs