Works well, could do with a clean up.