adding some code tests