Add magnet support