Added include statement and fixed a couple bugs