Code working fine