Damn, I really need to clean up the code-