Error fixing
Some brief late night coding to try to get back in the game