Now also working with the custom controller

Custom controller being the makeymakey