Changed controller design a little