Slight changes to code, was testing memory leak