Work on automated tests, should kinda work, need testing.