Adding automated tests