chore(cleaning up code)