Add progressDialog for better UI