improve the interface, now it check the case the user choosed