Added small test