Added a simple implementation of class