Added working code
Please enter the commit message for your changes. Lines starting