Cleaned unwanted lines of code and modified the get_vrf_id logic