- TODO: add more tests. Pretty basic currently.