Changed from define to class to be able to pass array