Nice way to test. Yeah I need to write tests and stuff