Vikramjeet Virk underwent rigorous training to prepare for his role in 'Agent'