Add custom loot functionality

- Might need some tweaking but works fine.