added small code changes