Add specs for new functionality