Functionality Fixed

Just needs output to be fixed