Added another test and set capacity