Todo: Write more tests...