Add problems tests