Implemented till return type

Next have to handle key type