Added ability to pass classes through props