added a few lines of code
bit more useful code