Unit Testing

Signed-off-by: Jannes <janniebun@gmail.com>