Added more code; fixes