"Discussed immigration with Modi. India will do what is right when it comes to taking back illegal immigrants," Trump said