Needing to add testing suite