Started adding pytests