Start adding snapshot tests