Ran codePro and removed unnecessary imports.
Edited test case to reflect changes from frontend. Added comments.