Add website model tests