Update for new inputs - Adapted to the new IFAETopFramework - Close issues #3 (closed) and #5 (closed)