Modified interface

Signed-off-by: JFRheault <jf.rheault.1@gmail.com>