Small bug fix, updated project