Add cell tap functionality back