Added API functionality

Seems like this would be required as well