add prompt and fix shake detection