Changed my part slightly