improving behavior
updating code