adding unit and stress tests