Add some query function and a button