Adding new code tests