Polished up the code, it's working again!