update to Input Stage Design