Adding Firebase code