Added a few more lines of code from the examples that project is based on.