Improve GUI after pick up weapon