added a little bit code