added leo's code

added leo's code