lets try to improve our code