Test works for the most part