Added updated tests

Just recloned my repo to get it up to date with master, created new pull request