Added import statements, copied existing class