Making a class for more flexibility.