Доделана логика обоих окон, подчистил код.