Added required code

Added required code