Tidy up
Wrapper for WARP loss MF