Lost some code and broke the rest. Need to fix.