added a little code