Added the provided code