Testing

Implemented basic unit testing