Added the correct inputs and outputs