Making penalties for pot use more mellow