Added a little piece of code