Add a simple class and unittest