Having issue while running code on dataframe