Не правильно вспомнил сюжет, но общую идею правильно.