First attempts at unit testing, not working yet