Updated function