Added code to pass unit tests