Added the required changes