Added rough testing setup