added own code
final tests