got to add set data function for testing