Removed print statements and tidied tests