forgot something. now the code is fully working in case one