added a few lines of code

only like 381 lines of code