Warnings improved

Signed-off-by: jayrambhia <jayrambhia777@gmail.com>