I've updated the code with some more jQuery that doesn't work