Adding tests as a part of repo