I found this solution but it's now not working now (I am getting a fatal error), can anyone please update the code