Expanding coverage test to 88.3% of the code - Testing controller