Change the logic to allow use input their ak,sk and public for test.