Updated to use the correct selector (id not class)