Still need to work on setReturn