Edited, tried to add AlertDialog