Wrote failing tests for part II