edited code
edited now