The code should now be working