Edit specs

Some changes I made after implementing code