Updating controller to work with our API (so hacky)

Need to fix this later but for now … meh.