add logic to separate file

ok