My Modified Code