Adding tests. Nothing works yet though